from sklearn.tree import DecisionTreeClassifier, export_graphviz
from sklearn.datasets import load_breast_cancer

cancer = load_breast_cancer()
X = cancer.data
y = cancer.target

tree = DecisionTreeClassifier(max_depth = 3, random_state = 10)
tree.fit(X,y)
export_graphviz(tree, out_file="cancer.dot",
                class_names=["malignant","benign"],
                feature_names=cancer.feature_names, filled=True)
